package middlewares

import (
	"github.com/gin-gonic/gin"
	"kuangstudy-mall/apis/goods-web/jtoken"
	"net/http"
	"strings"
)

/*
*
jwt的拦截器验证
*/
func JWTAuth() gin.HandlerFunc {
	return func(c *gin.Context) {
		// 我们这里jwt鉴权取头部信息 x-token 登录时回返回token信息 这里前端需要把token存储到cookie或者本地localSstorage中 不过需要跟后端协商过期时间 可以约定刷新令牌或者重新登录
		token := c.Request.Header.Get("x-token")
		if token == "" {
			c.JSON(http.StatusUnauthorized, map[string]any{
				"code": 601,
				"data": "",
				"msg":  "请登录",
			})
			// 挂起
			c.Abort()
			return
		}
		token = strings.Split(token, " ")[1]
		j := jtoken.NewJWT()
		// parseToken 解析token包含的信息
		claims, err := j.ParseToken(token)
		if err != nil {
			if err == jtoken.TokenExpired {
				if err == jtoken.TokenExpired {
					// 可以考虑续期或者刷新token
					c.JSON(http.StatusUnauthorized, map[string]any{
						"code": 601,
						"data": "",
						"msg":  "授权已过期",
					})
					c.Abort()
					return
				}
			}

			c.JSON(http.StatusUnauthorized, "未登陆")
			c.Abort()
			return
		}
		c.Set("claims", claims)
		c.Set("userId", claims.ID)
		c.Set("userName", claims.NickName)
		c.Set("role", claims.AuthorityId)
		c.Next()
	}
}
